package org.jeecg.config.jimureport;

import cn.hutool.core.bean.BeanUtil;
import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.jeecg.common.system.util.JwtUtil;
import org.jeecg.common.system.vo.SysUserCacheInfo;
import org.jeecg.common.util.RedisUtil;
import org.jeecg.common.util.TokenUtils;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.common.utils.BeanObjUtil;
import org.jeecg.modules.common.utils.StringUtil;
import org.jeecg.modules.jmreport.api.JmReportTokenServiceI;
import org.jeecg.modules.system.service.impl.SysBaseApiImpl;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

/**
 * 自定义积木报表鉴权(如果不进行自定义，则所有请求不做权限控制)
 * * 1.自定义获取登录token
 * * 2.自定义获取登录用户
 *
 * @author: baochangrong
 */
@Slf4j
@Component
public class JimuReportTokenService implements JmReportTokenServiceI {
    @Autowired
    private SysBaseApiImpl sysBaseApi;
    @Autowired
    @Lazy
    private RedisUtil redisUtil;

    private Map urlParamObject = new HashMap();

    @Override
    public String getToken(HttpServletRequest request) {
        //获取url 参数 2022年7月21号添加
        setHeardParams(request);
        return TokenUtils.getTokenByRequest(request);
    }

    private void setHeardParams(HttpServletRequest request) {
        Enumeration<String> referer = request.getHeaders("referer");
        String url = null;
        while (referer.hasMoreElements()) {
            url = (String) referer.nextElement();
            System.out.println(url);
            if (url.indexOf("http://") != -1) {
                break;
            }
        }
        if (!StringUtil.isEmpty(url)) {
            if (url.indexOf("?") != -1) {
                String[] urlParam = url.substring(url.indexOf("?")+1).split("&");
                if (urlParam.length > 0) {
                    for (String str : urlParam) {
                        urlParamObject.put(str.split("=")[0], str.split("=")[1]);
                    }
                }
            }
        }
    }

    @Override
    public String getUsername(String token) {
        return JwtUtil.getUsername(token);
    }

    @Override
    public Boolean verifyToken(String token) {
        return TokenUtils.verifyToken(token, sysBaseApi, redisUtil);
    }

    @Override
    public Map<String, Object> getUserInfo(String token) {
        Map<String, Object> map = new HashMap(5);
        String username = JwtUtil.getUsername(token);
        //此处通过token只能拿到一个信息 用户账号  后面的就是根据账号获取其他信息 查询数据或是走redis 用户根据自身业务可自定义
        SysUserCacheInfo userInfo = null;
        try {
            userInfo = sysBaseApi.getCacheUser(username);
        } catch (Exception e) {
            log.error("获取用户信息异常:" + e.getMessage());
            return map;
        }
        //设置账号名 sysUserCode
        map.put(SYS_USER_CODE, userInfo.getSysUserCode());
        //设置部门编码 sysOrgCode
        map.put(SYS_ORG_CODE, userInfo.getSysOrgCode());
        // 将所有信息存放至map 解析sql/api会根据map的键值解析
        //报表设计系统变量： 学年 schoolYearId ，学期 semesterId ，租户 tenantId ，用户 sysUserCode ，部门 sysOrgCode，当前日期 sysDate ，当前时间  sysDateTime
        map.put("tenantId", userInfo.getTenantId());//oConvertUtils.isEmpty(userInfo.getTenantId())? tenantId : userInfo.getTenantId());
        BeanObjUtil.merageMap(map, urlParamObject);

        return map;
    }
}
